package com.project.base.common;

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.ParseException;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.entity.StringEntity;
import org.apache.http.entity.mime.HttpMultipartMode;
import org.apache.http.entity.mime.MultipartEntity;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;
import org.apache.http.util.EntityUtils;
import org.apache.log4j.Logger;

import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.nio.charset.Charset;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;


/**
 * @author admin
 * @ClassName: HttpsUtil
 * @Description: Http帮助类
 * @date 2014-11-8 上午9:58:54
 */
public class HttpsUtil {

    private static final Logger LOG = Logger.getLogger(HttpsUtil.class);

    public static final String CHARACTER_ENCODING = "UTF-8";
    public static final String PATH_SIGN = "/";
    public static final String METHOD_POST = "POST";
    public static final String METHOD_GET = "GET";
    public static final String CONTENT_TYPE = "Content-Type";
    public static final String USER_AGENT = "User-Agent";

    public static final int CONNECTION_TIMEOUT = 10000;
    public static final int SO_TIMEOUT = 30000;
    public static final int SO_TIMEOUT_60S = 60000;

    public static final int BUFFER = 1024;

    /**
     * HttpResponse包装类
     *
     * @author qiuguobin
     */
    public static class HttpResponseWrapper {
        private HttpResponse httpResponse;
        private HttpClient httpClient;

        public HttpResponseWrapper(HttpClient httpClient, HttpResponse httpResponse) {
            this.httpClient = httpClient;
            this.httpResponse = httpResponse;
        }

        public HttpResponseWrapper(HttpClient httpClient) {
            this.httpClient = httpClient;
        }

        public HttpResponse getHttpResponse() {
            return httpResponse;
        }

        public void setHttpResponse(HttpResponse httpResponse) {
            this.httpResponse = httpResponse;
        }

        /**
         * 获得流类型的响应
         */
        public InputStream getResponseStream() throws IllegalStateException, IOException {
            return httpResponse.getEntity().getContent();
        }

        /**
         * 获得字符串类型的响应
         */
        public String getResponseString(String responseCharacter) throws ParseException, IOException {
            HttpEntity entity = getEntity();
            String responseStr = EntityUtils.toString(entity, responseCharacter);
            if (entity.getContentType() == null) {
                responseStr = new String(responseStr.getBytes("iso-8859-1"), responseCharacter);
            }
            EntityUtils.consume(entity);
            return responseStr;
        }

        public String getResponseString() throws ParseException, IOException {
            return getResponseString(CHARACTER_ENCODING);
        }

        /**
         * 获得响应状态码
         */
        public int getStatusCode() {
            return httpResponse.getStatusLine().getStatusCode();
        }

        /**
         * 获得响应状态码并释放资源
         */
        public int getStatusCodeAndClose() {
            close();
            return getStatusCode();
        }

        public HttpEntity getEntity() {
            return httpResponse.getEntity();
        }

        /**
         * 释放资源
         */
        public void close() {
            httpClient.getConnectionManager().shutdown();
        }
    }

    /**
     * POST方式提交上传文件请求
     */
    public static HttpResponseWrapper requestPostUpload(String url, Map<String, File> requestFiles, Map<String, String> requestParas, String requestCharacter) throws ClientProtocolException, IOException {
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient();
        } else {
            client = createHttpClient();
        }
        HttpPost httpPost = new HttpPost(url);
        httpPost.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        MultipartEntity multipartEntity = new MultipartEntity(HttpMultipartMode.BROWSER_COMPATIBLE, null, Charset.forName(requestCharacter));
        if (requestFiles == null || requestFiles.size() == 0) {
            return null;
        }
        for (Map.Entry<String, File> entry : requestFiles.entrySet()) {
            multipartEntity.addPart(entry.getKey(), new FileBody(entry.getValue(), "application/octet-stream", requestCharacter));
        }
        if (requestParas != null && requestParas.size() > 0) {
            // 对key进行排序
            List<String> keys = new ArrayList<String>(requestParas.keySet());
            Collections.sort(keys);
            for (String key : keys) {
                multipartEntity.addPart(key, new StringBody(requestParas.get(key), Charset.forName(requestCharacter)));
            }
        }
        httpPost.setEntity(multipartEntity);
        HttpResponse httpResponse = client.execute(httpPost);
        return new HttpResponseWrapper(client, httpResponse);
    }

    /**
     * POST方式提交上传文件请求
     */
    public static HttpResponseWrapper requestPostUpload(String url, Map<String, File> requestFiles, Map<String, String> requestParas) throws ClientProtocolException, IOException {
        return requestPostUpload(url, requestFiles, requestParas, CHARACTER_ENCODING);
    }

    /**
     * POST方式提交表单数据，返回响应对象
     */
    public static HttpResponseWrapper requestPostFormResponse(String url, Map<String, String> requestParas, String requestCharacter) throws ClientProtocolException, IOException {
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient();
        } else {
            client = createHttpClient();
        }
        HttpPost httpPost = new HttpPost(url);
        httpPost.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        List<NameValuePair> formParams = initNameValuePair(requestParas);
        httpPost.setEntity(new UrlEncodedFormEntity(formParams, requestCharacter));
        HttpResponse httpResponse = client.execute(httpPost); //执行POST请求
        return new HttpResponseWrapper(client, httpResponse);
    }


    /**
     * POST方式提交表单数据，返回响应对象，utf8编码
     */
    public static HttpResponseWrapper requestPostFormResponse(String url, Map<String, String> requestParas) throws ClientProtocolException, IOException {
        return requestPostFormResponse(url, requestParas, CHARACTER_ENCODING);
    }

    /**
     * POST方式提交表单数据，不会自动重定向
     */
    public static String requestPostForm(String url, Map<String, String> requestParas, String requestCharacter, String responseCharacter) {
        HttpResponseWrapper httpResponseWrapper = null;
        try {
            httpResponseWrapper = requestPostFormResponse(url, requestParas, requestCharacter);
            return httpResponseWrapper.getResponseString(responseCharacter);
        } catch (Exception e) {
            LOG.error(e);

        } finally {
            httpResponseWrapper.close();
        }
        return null;

    }

    /**
     * POST方式提交表单数据，不会自动重定向
     */
    public static String requestPostForm(String url, Map<String, String> requestParas) {
        return requestPostForm(url, requestParas, CHARACTER_ENCODING, CHARACTER_ENCODING);
    }


    public static HttpResponseWrapper requestGetResponse(String url) throws ClientProtocolException, IOException {
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient();
        } else {
            client = createHttpClient();
        }
        HttpGet httpGet = new HttpGet(url);
        httpGet.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        HttpResponse httpResponse = client.execute(httpGet);
        return new HttpResponseWrapper(client, httpResponse);
    }

    public static HttpResponseWrapper requestGetResponse(String url, String xForWardedFor) throws ClientProtocolException, IOException {
        LOG.info("URL: " + url);
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient();
        } else {
            client = createHttpClient();
        }
        HttpGet httpGet = new HttpGet(url);
        httpGet.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        if (StringTools.isValid(xForWardedFor)) {
            httpGet.addHeader("X-Forwarded-For", xForWardedFor);
        }
        HttpResponse httpResponse = client.execute(httpGet);
        return new HttpResponseWrapper(client, httpResponse);
    }

    /**
     * GET方式提交URL请求，会自动重定向
     */
    public static String requestGet(String url, String responseCharacter, String xForWardedFor) {
        HttpResponseWrapper httpResponseWrapper = null;
        try {
            httpResponseWrapper = requestGetResponse(url, xForWardedFor);
            return httpResponseWrapper.getResponseString(responseCharacter);
        } catch (Exception e) {
            LOG.error(e);
        } finally {
            if (httpResponseWrapper != null) {
                httpResponseWrapper.close();
            }
        }
        return null;
    }

    /**
     * GET方式提交URL请求，会自动重定向
     */
    public static String requestGet(String url) {
        return requestGet(url, CHARACTER_ENCODING, null);
    }

    /**
     * get请求
     *
     * @param url    请求url(eg:https://api.lvmama.com/user/getUserInfo)
     * @param params 参数
     * @return 响应信息
     */
    public static String requestGet(String url, Map<String, Object> params) {
        StringBuilder urlBuilder = new StringBuilder(url);
        urlBuilder.append("?").append(mapToUrl(params));
        return requestGet(urlBuilder.toString(), CHARACTER_ENCODING, null);
    }

    /**
     * 将Map中的数据组装成url
     *
     * @param params
     * @return
     */
    public static String mapToUrl(Map<String, Object> params) {
        StringBuilder sb = new StringBuilder();
        try {
            boolean isFirst = true;
            for (String key : params.keySet()) {
                String value = (String) params.get(key);
                if (isFirst) {
                    sb.append(key + "=" + URLEncoder.encode(value, "utf-8"));
                    isFirst = false;
                } else {
                    if (value != null) {
                        sb.append("&" + key + "="
                                + URLEncoder.encode(value, "utf-8"));
                    } else {
                        sb.append("&" + key + "=");
                    }
                }
            }
        } catch (Exception e) {
            LOG.error(e);
        }
        return sb.toString();
    }

    /**
     * GET方式提交URL请求，会自动重定向
     */
    public static String proxyRequestGet(String url, String xForWardedFor) {
        return requestGet(url, CHARACTER_ENCODING, xForWardedFor);
    }

    /**
     * POST方式提交非表单数据，返回响应对象
     */
    public static HttpResponseWrapper requestPostData(String url, String data, String contentType, String requestCharacter, int connectionTimeout, int soTimeout) throws ClientProtocolException, IOException {
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient(connectionTimeout, soTimeout);
        } else {
            client = createHttpClient(connectionTimeout, soTimeout);
        }
        HttpPost httpPost = new HttpPost(url);
        httpPost.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        httpPost.setHeader(CONTENT_TYPE, contentType);
        httpPost.setEntity(new StringEntity(data, requestCharacter));
        HttpResponse httpResponse = client.execute(httpPost);
        return new HttpResponseWrapper(client, httpResponse);
    }

    /**
     * POST方式提交非表单数据，返回响应对象
     */
    public static HttpResponseWrapper requestPostData(String url, String data, String contentType, String requestCharacter) throws ClientProtocolException, IOException {
        return requestPostData(url, data, contentType, requestCharacter, CONNECTION_TIMEOUT, SO_TIMEOUT);
    }

    /**
     * POST非表单方式提交XML数据
     */
    public static String requestPostXml(String url, String xmlData, String requestCharacter, String responseCharacter) {
        HttpResponseWrapper httpResponseWrapper = null;
        try {
            String contentType = "text/xml; charset=" + requestCharacter;
            httpResponseWrapper = requestPostData(url, xmlData, contentType, requestCharacter);
            return httpResponseWrapper.getResponseString(responseCharacter);
        } catch (Exception e) {
            LOG.error(e);
        } finally {
            httpResponseWrapper.close();
        }
        return null;
    }

    /**
     * POST非表单方式提交XML数据
     */
    public static String requestPostXml(String url, String xmlData) {
        return requestPostXml(url, xmlData, CHARACTER_ENCODING, CHARACTER_ENCODING);
    }

    public static HttpClient createHttpClient(int connectionTimeout, int soTimeout) {
        HttpClient httpClient = new DefaultHttpClient();
        HttpParams params = httpClient.getParams();
        HttpConnectionParams.setConnectionTimeout(params, connectionTimeout);
        HttpConnectionParams.setSoTimeout(params, soTimeout);
        return httpClient;
    }

    public static HttpClient createHttpClient() {
        return createHttpClient(CONNECTION_TIMEOUT, SO_TIMEOUT);
    }


    @SuppressWarnings("deprecation")
    public static HttpClient createHttpsClient(int connectionTimeout, int soTimeout) {
        try {
            CloseableHttpClient httpClient = new DefaultHttpClient();
            ;

            HttpParams params = httpClient.getParams();
            HttpConnectionParams.setConnectionTimeout(params, connectionTimeout);
            HttpConnectionParams.setSoTimeout(params, soTimeout);
            //TLS1.0与SSL3.0基本上没有太大的差别，可粗略理解为TLS是SSL的继承者，但它们使用的是相同的SSLContext
            SSLContext ctx = SSLContext.getInstance("TLS");
            //使用TrustManager来初始化该上下文，TrustManager只是被SSL的Socket所使用
            ctx.init(null, new TrustManager[]{new TrustAnyTrustManager()}, null);
            //创建SSLSocketFactory
            SSLSocketFactory socketFactory = new SSLSocketFactory(ctx);
            //通过SchemeRegistry将SSLSocketFactory注册到我们的HttpClient上
            httpClient.getConnectionManager().getSchemeRegistry().register(new Scheme("https", 443, socketFactory));
            return httpClient;
        } catch (Exception e) {
            LOG.error(e);
        }
        return null;
    }

    public static HttpClient createHttpsClient() {
        return createHttpsClient(CONNECTION_TIMEOUT, SO_TIMEOUT);
    }


    public static List<NameValuePair> initNameValuePair(Map<String, String> params) {
        List<NameValuePair> formParams = new ArrayList<NameValuePair>();
        if (params != null && params.size() > 0) {
            // 对key进行排序
            List<String> keys = new ArrayList<String>(params.keySet());
            Collections.sort(keys);
            for (String key : keys) {
                //LOG.info(key+" = " +params.get(key));
                formParams.add(new BasicNameValuePair(key, params.get(key)));
            }
        }
        return formParams;
    }


    private static class TrustAnyTrustManager implements X509TrustManager {

        public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {
        }

        public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {
        }

        public X509Certificate[] getAcceptedIssuers() {
            return new X509Certificate[]{};
        }
    }

    /******************** 已下为客户端新加  2014-05-29*****************/
    /**
     * 客户端调用
     * POST方式提交表单数据，不会自动重定向
     *
     * @param url           请求url
     * @param requestParas  请求参数
     * @param xForWardedFor 请求ip（统计客户端登陆次数）
     */
    public static String requestPostForm(String url, Map<String, String> requestParas, String xForWardedFor) {
        return requestPostForm(url, requestParas, xForWardedFor, CHARACTER_ENCODING, CHARACTER_ENCODING);
    }

    /**
     * 客户端登陆调用
     *
     * @param url               请求url
     * @param requestParas      请求参数
     * @param xForWardedFor     请求ip（统计客户端登陆次数）
     * @param requestCharacter  请求编码
     * @param responseCharacter 响应编码
     * @return
     */
    public static String requestPostForm(String url, Map<String, String> requestParas, String xForWardedFor, String requestCharacter, String responseCharacter) {
        HttpResponseWrapper httpResponseWrapper = null;
        try {
            httpResponseWrapper = requestPostFormResponse(url, requestParas, requestCharacter, xForWardedFor);
            return httpResponseWrapper.getResponseString(responseCharacter);
        } catch (Exception e) {
            LOG.error(e);
        } finally {
            httpResponseWrapper.close();
        }
        return null;
    }

    /**
     * POST方式提交表单数据，返回响应对象
     *
     * @param url           请求url
     * @param requestParas  请求参数
     * @param xForWardedFor 请求ip（统计客户端登陆次数）
     */
    public static HttpResponseWrapper requestPostFormResponse(String url, Map<String, String> requestParas, String requestCharacter, String xForWardedFor) throws ClientProtocolException, IOException {
        HttpClient client = null;
        if (url.startsWith("https")) {
            client = createHttpsClient();
        } else {
            client = createHttpClient();
        }
        HttpPost httpPost = new HttpPost(url);
        httpPost.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
        if (StringTools.isValid(xForWardedFor)) {
            httpPost.addHeader("X-Forwarded-For", xForWardedFor);
        }
        List<NameValuePair> formParams = initNameValuePair(requestParas);
        httpPost.setEntity(new UrlEncodedFormEntity(formParams, requestCharacter));
        HttpResponse httpResponse = client.execute(httpPost); //执行POST请求
        return new HttpResponseWrapper(client, httpResponse);
    }

    public static HttpClient createHttpClient(Map<String, Object> httpParams) {
        HttpClient httpClient = new DefaultHttpClient();
        HttpParams params = httpClient.getParams();
        if (params != null && httpParams != null) {
            for (String key : httpParams.keySet()) {
                params.setParameter(key, httpParams.get(key));
            }
        }
        return httpClient;
    }

    public static HttpClient createHttpsClient(Map<String, Object> httpParams) {
        try {
            HttpClient httpClient = new DefaultHttpClient(); //创建默认的httpClient实例
            HttpParams params = httpClient.getParams();
            if (params != null && httpParams != null) {
                for (String key : httpParams.keySet()) {
                    params.setParameter(key, httpParams.get(key));
                }
            }
            //TLS1.0与SSL3.0基本上没有太大的差别，可粗略理解为TLS是SSL的继承者，但它们使用的是相同的SSLContext
            SSLContext ctx = SSLContext.getInstance("TLS");
            //使用TrustManager来初始化该上下文，TrustManager只是被SSL的Socket所使用
            ctx.init(null, new TrustManager[]{new TrustAnyTrustManager()}, null);
            //创建SSLSocketFactory
            SSLSocketFactory socketFactory = new SSLSocketFactory(ctx);
            //通过SchemeRegistry将SSLSocketFactory注册到我们的HttpClient上
            httpClient.getConnectionManager().getSchemeRegistry().register(new Scheme("https", 443, socketFactory));
            return httpClient;
        } catch (Exception e) {
            LOG.error(e);
        }
        return null;
    }

    /**
     * @param url        请求url
     * @param params     请求参数
     * @param httpParams 设置HttpClient.getParams()中的参数
     * @return String
     * @autor: heyuxing  2014-11-14 下午8:02:50
     */
    public static String requestGet(String url, Map<String, Object> params, Map<String, Object> httpParams) {
        return requestGet(url, params, httpParams, null);
    }

    /**
     * @param url               请求url
     * @param params            请求参数
     * @param httpParams        设置HttpClient.getParams()中的参数
     * @param responseCharacter 处理应答response字符串使用的字符集
     * @return String
     * @autor: heyuxing  2014-12-11 下午2:29:01
     */
    public static String requestGet(String url, Map<String, Object> params, Map<String, Object> httpParams, String responseCharacter) {
        StringBuilder urlBuilder = new StringBuilder(url);
        urlBuilder.append("?").append(mapToUrl(params));

        HttpResponseWrapper httpResponseWrapper = null;
        try {
            HttpClient client = null;
            if (url.startsWith("https")) {
                client = createHttpsClient(httpParams);
            } else {
                client = createHttpClient(httpParams);
            }
            HttpGet httpGet = new HttpGet(urlBuilder.toString());
            httpGet.setHeader(USER_AGENT, "WAP_LVMM|ANDROID_|IPHONE_|WP8_");
            HttpResponse httpResponse = client.execute(httpGet);

            if (!StringTools.isValid(responseCharacter)) {
                responseCharacter = CHARACTER_ENCODING;
            }
            return new HttpResponseWrapper(client, httpResponse).getResponseString(responseCharacter);
        } catch (Exception e) {
            LOG.error(e);
        } finally {
            if (httpResponseWrapper != null) {
                httpResponseWrapper.close();
            }
        }
        return null;
    }
}
